python program measure time for block of code